from torchtext.datasets import WikiText2

# 下载并加载 WikiText2 数据集
train_iter = WikiText2(split='train')
for item in train_iter:
    print(item)
    break
